package com.example.demo.util;

import org.apache.poi.hssf.usermodel.*;

import java.io.FileOutputStream;
import java.io.IOException;

/**
 * @Classname ExportUtil
 * @Description TODO
 * @Date 2019/8/12 10:43
 * @Created by yby
 */
public class ExportUtil {

    /**
     * @param
     * @return
     * @Description TODO
     * @Param
     * @date 2019/8/12 10:46
     * @author yby
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, String[] title, String[][] values, HSSFWorkbook wb) {

        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if (wb == null) {
            wb = new HSSFWorkbook();
        }

        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        //style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.length; i++) {
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);
        }

        //创建内容
        for (int i = 0; i < values.length; i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < values[i].length; j++) {
                //将内容按顺序赋给对应的列对象
                row.createCell(j).setCellValue(values[i][j]);
            }
        }
        return wb;
    }

    public static void main(String[] args) {
        //最简单的导出excle方法
        HSSFWorkbook hssfWorkbook = new HSSFWorkbook();//首先在方法里面创建 HSSFWorkbook对象：
        HSSFSheet hssfSheet = hssfWorkbook.createSheet("sheet");//对应Excel里面的sheet；
        HSSFRow row = hssfSheet.createRow(0);//0代表着第一行，1代表着第二行，以此类推；
        HSSFCell cell = row.createCell(0);//0代表本行的第一个单元格，1代表第二个，以此类推
        cell.setCellValue("test");//利用HSSFCell对象进行填充文本值
        try {
            FileOutputStream out = new FileOutputStream("D:\\新森\\Excle\\test.xls");

            hssfWorkbook.write(out);

            out.flush();
        } catch (IOException E) {

        }
    }
}
